#pragma once

#include <opencv2/opencv.hpp>

// 装甲板类
struct Armor {
  cv::RotatedRect bar1; // 灯条1
  cv::RotatedRect bar2; // 灯条2
  std::string type;     // 装甲板类型
  double confidence_level;
  cv::Mat tvec; // 位置
  cv::Mat rvec; // 方向

  Armor(const cv::RotatedRect &bar1, const cv::RotatedRect &bar2,
        const std::string &type = "Unknown", const double &confidence_level = 0,
        const cv::Mat &tvec = cv::Mat(), const cv::Mat &rvec = cv::Mat())
      : bar1(bar1), bar2(bar2), type(type), confidence_level(confidence_level),
        tvec(tvec), rvec(rvec) {}
};